package com.zhang.wanx.controller;

import com.zhang.wanx.common.model.GenerationResDTO;
import com.zhang.wanx.service.IWanXiangService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

/**
 * description
 *
 * @author zhangFanJun
 * @date 2024-03-26 17:32
 **/
@RequiredArgsConstructor
@Slf4j
@RequestMapping("/wanx")
@RestController
public class WanXiangController {

    private final IWanXiangService wanXiangService;

    /**
     * 生成任务
     * styleIndex：
     * 0 复古漫画
     * 1 3D童话
     * 2 二次元
     * 3 小清新
     * 4 未来科技
     * 5 国画古风
     * 6 将军百战
     * 7 炫彩卡通
     * 8 清雅国风
     * 9 喜迎新年
     *
     * @param file 文件
     * @return 结果
     * @author zfj
     * @date 2024/3/26
     */
    @PostMapping("/generation/{styleIndex}")
    public String generation(MultipartFile file,@PathVariable("styleIndex") Integer styleIndex){

        return wanXiangService.generation(file,styleIndex);
    }


    /**
     * 获取任务结果
     *
     * @param id 任务id
     * @return 结果
     * @author zfj
     * @date 2024/3/26
     */
    @GetMapping("/tasks/{id}")
    public GenerationResDTO tasks(@PathVariable("id") String id ){

        return wanXiangService.tasks(id);
    }


}
